Getting (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same